package cn.tedu.v4;

import java.io.*;
import java.net.*;
import java.util.*;

/**
 * Version 4 多人聊天室 ---消息转发功能
 * 建议：单线聊天，多人聊天，多人聊天+QQ模式 【消息转发功能】
 * 拓展功能：建议增加第三个版本 Version 3
 *
 * 消息转发步骤：服务器给所有的客户端写出内容，客户端之间相互收到内容
 * 1.集合存储所有写出的客户端---服务器端写出给所有客户端
 * 2.广播给所有客户端--broadcastMessage 广播留言
 *  写在ClientHandler上,必须同步方法
 * 3.将客户端输出流添加 allOut集合中【必须同步块】
 *   注意：写在ClientHandler的构造器--所有客户端具备功能：消息转发
 * 4.广播消息给所有客户端
 *  注意：写在ClientHandler的run---客户端写出的信息，读入方法转发出去
 * 5.客户端退出，从 allOut 结合中移除
 *   注意：写在ClientHandler的run---finally【一定移除链接】
 */

public class Server{
    private ServerSocket serverSocket;
    private static final Integer PORT=4545;
    //1.私有属性：集合存储1个客户端链接,发送文本，转发给其他用户
    private List<PrintWriter> allOut=new ArrayList<>();
    public Server() {
        try {
            serverSocket = new ServerSocket(PORT);
        } catch (BindException e) {
            System.out.println("端口号重复申请，建议换个端口号");
        } catch (IOException e) {
            System.out.println("申请端口号失败~~~");
        }
    }

    public void start() {
        try {
            while (true) {
                System.out.println("----等待用户端链接---");
                Socket socket = serverSocket.accept();
                new Thread(new ClientHandler(socket)).start();
            }
        } catch (IOException e) {
            System.out.println("服务器端通讯断开了");
        } finally {

            if (serverSocket!=null&&!serverSocket.isClosed()){
                try {
                    serverSocket.close();
                } catch (IOException e) {
                    System.out.println("服务器关闭失败");
                }
            }

        }
    }


    public static void main(String[] args)  {
        new Server().start();
    }


    //2.广播消息给所有客户端---功能：broadcastMessage 广播留言
    //2.1broadcastMessage功能：提供转发给客户端消息【加同步锁】
    //   理解：来一个客户端，只允许当前客户端转发消息
    //2.2String message 参数：转发消息---代表消息
    private synchronized void  broadcastMessage(String message){
        //2.3增强for： allOut 存储所有客户端
        //   【理解：来很多客户端，很对客户端遍历出来，消息转发出去】
        for (PrintWriter pw:allOut){//allOut保存所有对象客户端
            pw.println("消息广播："+message);//理解：消息转发出去

        }
    }

    private class ClientHandler implements Runnable {
        private Socket socket;
        private String ip;
        private BufferedReader br;
        private PrintWriter pw;
        public ClientHandler(Socket s){
            this.socket=s;
            ip=socket.getInetAddress().getHostAddress();

            try {
                br = new BufferedReader(new InputStreamReader(socket.getInputStream(), "UTF8"));
                pw=new PrintWriter(new BufferedWriter(new OutputStreamWriter(socket.getOutputStream(), "UTF8")), true);
                //3.所有客户端具备消息转发
                //  理解：客户端写出文字，服务器负责转发文字
                //监视同步对象：allOut---存所有客户端--
                //客户端存在线程中--执行存在执行顺序不一致情况
                synchronized (allOut) {
                    allOut.add(pw);//将客户端写出文字，存集合中
                }


            } catch (IOException e) {
                System.out.println("客户端断开通讯");
            }

        }

        public void run() {
            try {
                new Thread(()->{
                    try {
                        String message;
                        while ((message=br.readLine())!=null) {
                            System.out.println("客户端说："+message);
                            System.out.println("服务器端请输入：");
                            //4.广播消息给所有客户端
                            //broadcastMessage转发：写出文本内容
                            //message 客户端发给服务器的内容
                            //把当前客户端写的内容，通过服务器端读入：message
                            // 转发和给所有的客户端
                            broadcastMessage(message);
                        }
                    } catch (IOException e) {
                        System.out.println("客户端读入失败了");
                    }
                }).start();

                Scanner input = new Scanner(System.in);
                while (true) {
                    System.out.println("服务器请输入：");
                    String say=input.nextLine();
                    pw.println("服务器说："+say);
                }



                } catch(NullPointerException e){
                    System.out.println("IO流传入对象，有对象为null,立马核对，建议：查看try块中对象可能赋值为null");
                } finally{
                    //5.一旦有 客户端断开，客户都拿从 allOut 集合中移除  输出流
                    // 必须同步块，结束客户端不一致
                    synchronized (allOut){//监视allOut集合中客户都拿有没有断开
                        allOut.remove(pw);//allOut集合客户都拿断开连接，移除写出流
                    }
                    if (br != null)
                        try {
                            br.close();
                        } catch (IOException e) {
                            System.out.println("结束通讯：关闭失败");
                        }

                    if (pw!=null) pw.close();
                    if (socket!=null&&!socket.isClosed()){
                        try {
                            socket.close();
                        } catch (IOException e) {
                            System.out.println("通讯关闭失败");
                        }
                    }
                }
            }
        }
    }


